package leetcode;

/**
 * ClassName: MyLinkedListNode
 * Package: leetcode
 * Description:
 *
 * @Author wjn
 * @Create 2025/5/27 22:02
 * @Version 1.0
 */
public class MyLinkedList {

    //size存储链表元素的个数
    private int size;
    //注意这里记录的是虚拟头结点
    private ListNode dummyHead;

    public MyLinkedList() {
        this.size = 0;
        this.dummyHead = new ListNode(0);
    }

    /**
     * 获取链表中第 index 个节点的值。如果索引无效，则返回-1。
     *
     * @param index
     * @return
     */
    public int get(int index) {
        //如果index非法，返回-1
        if (index < 0 || index >= size) {
            return -1;
        }
        ListNode cur = dummyHead.next;
        //第0个节点是虚拟头节点，所以查找第 index 个节点
        for (int i = 0; i < index; i++) {
            cur = cur.next;
        }
        return cur.val;
    }

    /**
     * 在链表的第一个元素之前添加一个值为 val 的节点。插入后，新节点将成为链表的第一个节点。
     *
     * @param val
     */
    public void addAtHead(int val) {
        ListNode newNode = new ListNode(val);
        // 下面的顺序很重要
        newNode.next = dummyHead.next;
        dummyHead.next = newNode;
        size++;

    }

    public void addAtTail(int val) {
        ListNode newNode = new ListNode(val);
        ListNode cur = dummyHead;
        while (cur.next != null) {
            cur = cur.next;
        }
        cur.next = newNode;
        size++;
    }

    public void addAtIndex(int index, int val) {
        if (index < 0 || index > size) {
            return;
        }
        ListNode pre = dummyHead;
        //找到前置节点
        for (int i = 0; i < index; i++) {
            pre = pre.next;
        }
        ListNode newNode = new ListNode(val);
        // 顺序很重要
        newNode.next = pre.next;
        pre.next = newNode;
        size++;
    }

    public void deleteAtIndex(int index) {
        if (index < 0 || index >= size) {
            return;
        }
        //因为有虚拟头节点，所以不用对index=0的情况进行特殊处理
        ListNode pre = dummyHead;
        //找到前置节点
        for (int i = 0; i < index ; i++) {
            pre = pre.next;
        }
        //删除
        pre.next = pre.next.next;
        size--;
    }
}
